Skip to content

[CMSIS-NN] Removed batch_size assert in depthwise_conv#3562

Merged
veblush merged 1 commit into
tensorflow:mainfrom
veblush:cm-dw
May 27, 2026
Merged

[CMSIS-NN] Removed batch_size assert in depthwise_conv#3562
veblush merged 1 commit into
tensorflow:mainfrom
veblush:cm-dw

Conversation

@veblush
Copy link
Copy Markdown
Collaborator

@veblush veblush commented May 21, 2026

Since CMSIS-NN actually supports batch sizes > 1 natively (it loops over input_batches internally), this assertion seems to be outdated or unnecessarily restrictive. Therefore, I removed the batch_size == 1 check.

BUG=N/A

@veblush veblush requested a review from a team as a code owner May 21, 2026 18:06
@veblush veblush changed the title Removed batch_size assert in depthwise_conv [CMSIS-NN Removed batch_size assert in depthwise_conv May 21, 2026
@veblush veblush changed the title [CMSIS-NN Removed batch_size assert in depthwise_conv [CMSIS-NN] Removed batch_size assert in depthwise_conv May 21, 2026
@veblush veblush added the ci:full Triggers the comprehensive cross-platform test suite. label May 21, 2026
@veblush veblush requested a review from suleshahid May 21, 2026 21:42
@veblush veblush added ci:full Triggers the comprehensive cross-platform test suite. and removed ci:full Triggers the comprehensive cross-platform test suite. labels May 21, 2026
@veblush veblush added this pull request to the merge queue May 27, 2026
Merged via the queue into tensorflow:main with commit cc99d05 May 27, 2026
41 checks passed
@veblush veblush deleted the cm-dw branch May 27, 2026 17:16
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ci:full Triggers the comprehensive cross-platform test suite.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants